#pragma once
#include "ControlBase.h"

class RadioBox : public Control
{
	float last_width = 0.0f;

public:
	virtual UIClass Type() { return UIClass::UI_CheckBox; }
	bool IsContainer() override
	{
		return false;
	}
	D2D1_COLOR_F UnderMouseColor = Colors::DarkSlateGray;
	float Boder = 1.5f;
	RadioBox(std::wstring text, int x, int y)
	{
		this->Text = text;
		this->Location = POINT{x, y};
		this->BackColor = D2D1_COLOR_F{0.75f, 0.75f, 0.75f, 0.75f};
	}
	SIZE ActualSize() override
	{
		auto d2d = this->Render;
		auto font = this->Font ? this->Font : d2d->DefaultFontObject;
		auto text_size = font->GetTextSize(this->Text);
		return SIZE{(int)text_size.width + int(text_size.height + 2), (int)text_size.height};
	}
	void Update() override
	{
		if (this->IsVisual == false)
			return;
		bool isUnderMouse = this->MostParent->UnderMouse == this;
		bool isSelected = this->MostParent->Selected == this;
		auto d2d = this->Render;
		auto abslocation = this->AbsLocation;
		auto size = this->ActualSize();
		auto absRect = this->AbsRect;
		if (last_width > size.cx)
		{
			absRect.right += last_width - size.cx;
			size.cx = last_width;
		}
		d2d->PushDrawRect(absRect.left, absRect.top, absRect.right - absRect.left, absRect.bottom - absRect.top);
		{
			Control *tmpc = this;
			List<Control *> need_render_background;
			while (tmpc->BackColor.a < 1.0f)
			{
				tmpc = tmpc->Parent;
				need_render_background.Add(tmpc);
			}
			if (this->BackColor.a < 1.0f)
			{
				for (int i = need_render_background.Count - 1; i >= 0; i--)
				{
					d2d->FillRect(abslocation.x, abslocation.y, size.cx, size.cy, need_render_background[i]->BackColor);
					if (need_render_background[i]->Image)
					{
						need_render_background[i]->RenderImage();
					}
				}
			}
			auto col = this->ForeColor;
			if (isUnderMouse)
			{
				col = UnderMouseColor;
			}
			auto font = this->Font ? this->Font : d2d->DefaultFontObject;
			auto textSize = font->GetTextSize(this->Text);

			d2d->DrawString(this->Text, abslocation.x + textSize.height + 2, abslocation.y, col);
			d2d->DrawEllipse(
				abslocation.x + (textSize.height * 0.5), abslocation.y + (textSize.height * 0.5),
				textSize.height * 0.3, textSize.height * 0.3,
				col);
			if (this->Checked)
			{
				d2d->FillEllipse(
					abslocation.x + (textSize.height * 0.5), abslocation.y + (textSize.height * 0.5),
					textSize.height * 0.2, textSize.height * 0.2,
					col);
			}
		}
		d2d->PopDrawRect();
		last_width = size.cx;
	}

	bool ProcessMessage(UINT message, WPARAM wParam, LPARAM lParam, int xof, int yof) override
	{
		switch (message)
		{
		case WM_DROPFILES:
		{
			HDROP hDropInfo = HDROP(wParam);
			UINT uFileNum = DragQueryFile(hDropInfo, 0xffffffff, NULL, 0);
			TCHAR strFileName[MAX_PATH];
			List<std::wstring> files;
			for (UINT i = 0; i < uFileNum; i++)
			{
				DragQueryFile(hDropInfo, i, strFileName, MAX_PATH);
				files.Add(strFileName);
			}
			DragFinish(hDropInfo);
			if (files.Count > 0)
			{
				this->OnDropFile(this, files);
			}
		}
		break;
		case WM_MOUSEWHEEL: // mouse wheel
		{
			MouseEventArgs event_obj = MouseEventArgs(MouseButtons::MouseButtons_None, 0, xof, yof, GET_WHEEL_DELTA_WPARAM(wParam));
			this->OnMouseWheel(this, event_obj);
		}
		break;
		case WM_MOUSEMOVE: // mouse move
		{
			MouseEventArgs event_obj = MouseEventArgs(MouseButtons::MouseButtons_None, 0, xof, yof, HIWORD(wParam));
			this->OnMouseMove(this, event_obj);
		}
		break;
		case WM_LBUTTONDOWN: // mouse down
		case WM_RBUTTONDOWN:
		case WM_MBUTTONDOWN:
		{
			if (WM_LBUTTONDOWN == message)
			{
				auto lastSelected = this->MostParent->Selected;
				this->MostParent->Selected = this;
				if (lastSelected && lastSelected != this)
				{
					lastSelected->SingleUpdate();
				}
			}
			MouseEventArgs event_obj = MouseEventArgs(FromParamToMouseButtons(message), 0, xof, yof, HIWORD(wParam));
			this->OnMouseDown(this, event_obj);
			this->SingleUpdate();
		}
		break;
		case WM_LBUTTONUP: // mouse up
		case WM_RBUTTONUP:
		case WM_MBUTTONUP:
		{
			if (WM_LBUTTONUP == message && this->MostParent->Selected == this)
			{
				if (this->Checked == false)
				{
					this->Checked = true;
					this->OnChecked(this);
				}
				MouseEventArgs event_obj = MouseEventArgs(FromParamToMouseButtons(message), 0, xof, yof, HIWORD(wParam));
				this->OnMouseClick(this, event_obj);
			}
			this->MostParent->Selected = NULL;
			MouseEventArgs event_obj = MouseEventArgs(FromParamToMouseButtons(message), 0, xof, yof, HIWORD(wParam));
			this->OnMouseUp(this, event_obj);
			this->SingleUpdate();
		}
		break;
		case WM_LBUTTONDBLCLK: // mouse double click
		{
			auto lastSelected = this->MostParent->Selected;
			this->MostParent->Selected = this;
			if (lastSelected && lastSelected == this)
			{
				this->Checked = !this->Checked;
				MouseEventArgs event_obj = MouseEventArgs(FromParamToMouseButtons(message), 0, xof, yof, HIWORD(wParam));
				this->OnMouseDoubleClick(this, event_obj);
				this->SingleUpdate();
			}
		}
		break;
		case WM_KEYDOWN: // keyboard down
		{
			KeyEventArgs event_obj = KeyEventArgs((Keys)(wParam | 0));
			this->OnKeyDown(this, event_obj);
		}
		break;
		case WM_KEYUP: // keyboard up
		{
			KeyEventArgs event_obj = KeyEventArgs((Keys)(wParam | 0));
			this->OnKeyUp(this, event_obj);
		}
		break;
		}
		return true;
	}
};
